package com.ysh.smartplatform.websocket;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.*;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.handler.TextWebSocketHandler;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 袁胜虎 on 2022/9/13 11:27
 */
@Component
@EnableWebSocket
public class TestWebsocket  extends TextWebSocketHandler {
    private ObjectMapper objectMapper;

    @Autowired
    public void setObjectMapper(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }
    //保存客户端的map
    private static Map<String, WebSocketSession> allClients = new ConcurrentHashMap<String,WebSocketSession>();

    /**
     * 链接成功之后执行,链接之后需要区分是谁,做了一个简单的要求,地址的最后一位就是名字,我们用名字进行区分
     * @param session
     * @throws Exception
     */
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.err.println("TestWebscoket中的方法afterConnectionEstablished执行了!!!");
        String path = session.getUri().getPath();
        //我们约定最后一个/后面的就是用户名
        String username = path.substring(path.lastIndexOf("/") + 1);
        //我们把数据保存到属性中,方便后面获取
        session.getAttributes().put("name",username);
        //保存了会话
        allClients.put(username,session);
    }

    /**
     * 当发生异常时运行
     * @param session
     * @param exception
     * @throws Exception
     */
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        super.handleTransportError(session, exception);
        System.err.println("TestWebscoket中的方法handleTransportError执行了!!!");
    }

    /**
     *当关闭链接时运行
     * @param session
     * @param status
     * @throws Exception
     */
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        super.afterConnectionClosed(session, status);
        System.err.println("TestWebscoket中的方法afterConnectionClosed执行了!!!");
    }

    /**
     * 处理文本信息 就是客户端发送了一段文本信息到服务器
     * @param session
     * @param message
     * @throws Exception
     */
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        String payload = message.getPayload();
        System.err.println("TestWebscoket中的方法handleTextMessage执行了!!!"+payload);
        //找到目标收消息的人的session,在某个地方一定保存了这些session,并且还可以根据zhangsan,lisi,王五来查询
        //我们怎么知道他想发给谁?我们约定,发送的数据格式是json,,理由有一个字段代表了接收者的名字,我们用toUser来传递数据,msg代表要发送的数据
        Map map = objectMapper.readValue(payload, Map.class);
        //接收者
        String toUser = (String) map.get("toUser");
        String msg = (String) map.get("msg");
        //找到对方的会话
        WebSocketSession toSession = allClients.get(toUser);

        Object username = session.getAttributes().get("name");

        if (toSession != null && toSession.isOpen()){
            //向另外一方主动发送数据
            toSession.sendMessage(new TextMessage("收到来自于"+username+"的消息,内容是:"+msg));
        }else {
            session.sendMessage(new TextMessage("对方不在线"));
        }
    }

    /**
     * 心跳信息 正在发送心跳的会话
     * @param session
     * @param message
     * @throws Exception
     */
    @Override
    protected void handlePongMessage(WebSocketSession session, PongMessage message) throws Exception {
        super.handlePongMessage(session, message);
        System.err.println("TestWebscoket中的方法handlePongMessage执行了!!!");
    }


}
